USER_AGENTS = [
    "Mozilla/4.0 (compatible; MSIE 6.0; Windows NT 5.1; SV1; AcooBrowser; .NET CLR 1.1.4322; .NET CLR 2.0.50727)",
    "Mozilla/4.0 (compatible; MSIE 7.0; Windows NT 6.0; Acoo Browser; SLCC1; .NET CLR 2.0.50727; Media Center PC 5.0; .NET CLR 3.0.04506)",
    "Mozilla/4.0 (compatible; MSIE 7.0; AOL 9.5; AOLBuild 4337.35; Windows NT 5.1; .NET CLR 1.1.4322; .NET CLR 2.0.50727)",
    "Mozilla/5.0 (Windows; U; MSIE 9.0; Windows NT 9.0; en-US)",
    "Mozilla/5.0 (compatible; MSIE 9.0; Windows NT 6.1; Win64; x64; Trident/5.0; .NET CLR 3.5.30729; .NET CLR 3.0.30729; .NET CLR 2.0.50727; Media Center PC 6.0)",
    "Mozilla/5.0 (compatible; MSIE 8.0; Windows NT 6.0; Trident/4.0; WOW64; Trident/4.0; SLCC2; .NET CLR 2.0.50727; .NET CLR 3.5.30729; .NET CLR 3.0.30729; .NET CLR 1.0.3705; .NET CLR 1.1.4322)",
    "Mozilla/4.0 (compatible; MSIE 7.0b; Windows NT 5.2; .NET CLR 1.1.4322; .NET CLR 2.0.50727; InfoPath.2; .NET CLR 3.0.04506.30)",
    "Mozilla/5.0 (Windows; U; Windows NT 5.1; zh-CN) AppleWebKit/523.15 (KHTML, like Gecko, Safari/419.3) Arora/0.3 (Change: 287 c9dfb30)",
    "Mozilla/5.0 (X11; U; Linux; en-US) AppleWebKit/527+ (KHTML, like Gecko, Safari/419.3) Arora/0.6",
    "Mozilla/5.0 (Windows; U; Windows NT 5.1; en-US; rv:1.8.1.2pre) Gecko/20070215 K-Ninja/2.1.1",
    "Mozilla/5.0 (Windows; U; Windows NT 5.1; zh-CN; rv:1.9) Gecko/20080705 Firefox/3.0 Kapiko/3.0",
    "Mozilla/5.0 (X11; Linux i686; U;) Gecko/20070322 Kazehakase/0.4.5",
    "Mozilla/5.0 (X11; U; Linux i686; en-US; rv:1.9.0.8) Gecko Fedora/1.9.0.8-1.fc10 Kazehakase/0.5.6",
    "Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/535.11 (KHTML, like Gecko) Chrome/17.0.963.56 Safari/535.11",
    "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_7_3) AppleWebKit/535.20 (KHTML, like Gecko) Chrome/19.0.1036.7 Safari/535.20",
    "Opera/9.80 (Macintosh; Intel Mac OS X 10.6.8; U; fr) Presto/2.9.168 Version/11.52",
    "Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/536.11 (KHTML, like Gecko) Chrome/20.0.1132.11 TaoBrowser/2.0 Safari/536.11",
    "Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/537.1 (KHTML, like Gecko) Chrome/21.0.1180.71 Safari/537.1 LBBROWSER",
    "Mozilla/5.0 (compatible; MSIE 9.0; Windows NT 6.1; WOW64; Trident/5.0; SLCC2; .NET CLR 2.0.50727; .NET CLR 3.5.30729; .NET CLR 3.0.30729; Media Center PC 6.0; .NET4.0C; .NET4.0E; LBBROWSER)",
    "Mozilla/4.0 (compatible; MSIE 6.0; Windows NT 5.1; SV1; QQDownload 732; .NET4.0C; .NET4.0E; LBBROWSER)",
    "Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/535.11 (KHTML, like Gecko) Chrome/17.0.963.84 Safari/535.11 LBBROWSER",
    "Mozilla/4.0 (compatible; MSIE 7.0; Windows NT 6.1; WOW64; Trident/5.0; SLCC2; .NET CLR 2.0.50727; .NET CLR 3.5.30729; .NET CLR 3.0.30729; Media Center PC 6.0; .NET4.0C; .NET4.0E)",
    "Mozilla/5.0 (compatible; MSIE 9.0; Windows NT 6.1; WOW64; Trident/5.0; SLCC2; .NET CLR 2.0.50727; .NET CLR 3.5.30729; .NET CLR 3.0.30729; Media Center PC 6.0; .NET4.0C; .NET4.0E; QQBrowser/7.0.3698.400)",
    "Mozilla/4.0 (compatible; MSIE 6.0; Windows NT 5.1; SV1; QQDownload 732; .NET4.0C; .NET4.0E)",
    "Mozilla/4.0 (compatible; MSIE 7.0; Windows NT 5.1; Trident/4.0; SV1; QQDownload 732; .NET4.0C; .NET4.0E; 360SE)",
    "Mozilla/4.0 (compatible; MSIE 6.0; Windows NT 5.1; SV1; QQDownload 732; .NET4.0C; .NET4.0E)",
    "Mozilla/4.0 (compatible; MSIE 7.0; Windows NT 6.1; WOW64; Trident/5.0; SLCC2; .NET CLR 2.0.50727; .NET CLR 3.5.30729; .NET CLR 3.0.30729; Media Center PC 6.0; .NET4.0C; .NET4.0E)",
    "Mozilla/5.0 (Windows NT 5.1) AppleWebKit/537.1 (KHTML, like Gecko) Chrome/21.0.1180.89 Safari/537.1",
    "Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/537.1 (KHTML, like Gecko) Chrome/21.0.1180.89 Safari/537.1",
    "Mozilla/5.0 (iPad; U; CPU OS 4_2_1 like Mac OS X; zh-cn) AppleWebKit/533.17.9 (KHTML, like Gecko) Version/5.0.2 Mobile/8C148 Safari/6533.18.5",
    "Mozilla/5.0 (Windows NT 6.1; Win64; x64; rv:2.0b13pre) Gecko/20110307 Firefox/4.0b13pre",
    "Mozilla/5.0 (X11; Ubuntu; Linux x86_64; rv:16.0) Gecko/20100101 Firefox/16.0",
    "Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/537.11 (KHTML, like Gecko) Chrome/23.0.1271.64 Safari/537.11",
    "Mozilla/5.0 (X11; U; Linux x86_64; zh-CN; rv:1.9.2.10) Gecko/20100922 Ubuntu/10.10 (maverick) Firefox/3.6.10"
]

def get_ua():
    return random.choice(USER_AGENTS)

flag = 1

# http://www.tianyancha.com/v2/search/%E7%BE%8E%E8%81%94%E5%BE%AE%E7%94%B5%E5%AD%90.json
# url = 'http://www.tianyancha.com/search?key=%s&checkFrom=searchBox'
# for name in companys:
#     if not DB['search_index'].find_one({'name':name.replace('(','（').replace(')','）')}):
#         print(name)

#         if flag %2 ==0:
#             browser.get(url%parse.quote(name))
#         else:
#             browser_1.get(url%parse.quote(name))
#         flag += 1
#         time.sleep(random.randint(8,15))
# #        browser.save_screenshot('jpg/%s.png'%name)

# for name in companys:
#     org = DB['search_index'].find_one({'name':name.replace('(','（').replace(')','）')})
#     if org:
#         if not DB['company_index'].find_one({'_id':org.get('_id')}):
#             print(int(org.get('_id')))
#             if flag %2 == 0:
#                 browser.get('http://www.tianyancha.com/company/%s'%int(org.get('_id')))
#             else:
#                 browser_1.get('http://www.tianyancha.com/company/%s'%int(org.get('_id')))
#             flag += 1
#             print(name)
#             time.sleep(random.randint(8,16))
#  #       browser.save_screenshot('jpg/%s.png'%name)



# rests = [org for org in DB['holdingCompany'].find({})]
# for org in rests:
#     if not DB['company_index'].find_one({'_id':org.get('id')}):
#         if not org.get('id') or not isinstance(org.get('id'),int):
#             continue
#         print(int(org.get('id')))
#         if flag %2 == 0:
#             browser.get('http://www.tianyancha.com/company/%s'%int(org.get('id')))
#         else:
#             browser_1.get('http://www.tianyancha.com/company/%s'%int(org.get('id')))
#         flag += 1
#         print(org.get('name'))
#         time.sleep(random.randint(8,16))
# #flag = 1

#for org in DB['search_index'].find({}):
#    if not DB['company_index'].find_one({'_id':org.get('_id')}):
#        print(int(org.get('id')))
#        if flag %2 == 0:
#            browser.get('http://www.tianyancha.com/company/%s'%int(org.get('id')))
#        else:
#            browser_1.get('http://www.tianyancha.com/company/%s'%int(org.get('id')))
#        flag += 1
#        print(org.get('name'))
#        time.sleep(random.randint(10,20))



import pymongo
import time
import random
from selenium import webdriver
from selenium.webdriver.common.desired_capabilities import DesiredCapabilities
from urllib import parse



class Browsers:

    def __init__(self,):
        self.proxies = ['192.168.1.220:8080','192.168.1.220:8081','192.168.1.220:8082','192.168.1.220:8083']
        self.num = 0
        self.get_browser()
        

    def get_browser(self):
        self.browsers = []
        for my_proxy in self.proxies:
            proxy=webdriver.Proxy()
            proxy.http_proxy=my_proxy
            proxy.add_to_capabilities(webdriver.DesiredCapabilities.PHANTOMJS)
            dcap = dict(DesiredCapabilities.PHANTOMJS)
            dcap["phantomjs.page.settings.userAgent"] = get_ua()
            dcap["phantomjs.page.settings.loadImages"] = False
            browser=webdriver.PhantomJS(desired_capabilities=dcap)
            self.browsers.append(browser)

    def get(self,url,sleep_time=random.randint(20,40)):
        flag = self.num % len(self.browsers)
        browser = self.browsers[flag]
        browser.get(url)
        self.num += 1
        time.sleep(sleep_time)
        self.check_bs()
        return browser

    def check_bs(self):
        if self.num >4000:
            for bs in self.browsers:
                bs.quit()
            self.get_browser()



TABLE = pymongo.MongoClient('192.168.1.220', 29001).merge.merge_4
TP_DB = pymongo.MongoClient('192.168.1.220', 27017).tianyancha.temp
names = [each.get('full_name') for each in TABLE.find({},{'full_name':1})]
print(len(names))
TYC = pymongo.MongoClient('192.168.1.220', 27017).tianyancha.search_index
t_names = []
for name in names:
    name = name.replace('(','（').replace(')','）')
    if not TYC.find_one({'name':name}):
        if not TP_DB.find_one({'_id':name}):
            t_names.append(name)
            if len(t_names)>10:
                break
url = 'http://www.tianyancha.com/search?key=%s&checkFrom=searchBox'
browser = Browsers()
for name in t_names:
    bs = browser.get(url%parse.quote(name))
    if name in bs.page_source:
        TP_DB.insert_one({'_id':name})
        print('search ok : %s'%name)
    else:
        print('search false : %s'%name)

